/*
 * @lc app=leetcode.cn id=309 lang=cpp
 *
 * [309] 最佳买卖股票时机含冷冻期
 */

// @lc code=start
class Solution {
public:
    int maxProfit(vector<int>& prices) {
        //状态机DP问题
        //当前的状态有3种
        //0.当天卖出 1.冷冻期 2.买入状态
        //状态变换
        //0->1; 1->2; 2->2; 2->0; 1->1
        if(prices.size() == 0) return 0;
        int INF = 1e8;
        vector<vector<int>> dp(prices.size(),vector<int>(3, -INF));
        dp[0][1] = 0;
        dp[0][2] = -prices[0];
        for(int i = 1; i < prices.size(); i++){
            dp[i][0] = dp[i-1][2] + prices[i];
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]);
            dp[i][2] = max(dp[i-1][2], dp[i-1][1] - prices[i]);
        }
        return max(dp[prices.size() - 1][0], max(dp[prices.size()-1][1], dp[prices.size() - 1][2]));
    }
};
// @lc code=end

